package com.yt.generator.engine.config;

import lombok.Data;
import org.apache.commons.lang3.StringUtils;

import java.util.ArrayList;
import java.util.List;

/**
 * Service模板生成的配置
 *
 * @author fengshuonan
 * @date 2017-05-07 22:12
 */
@Data
public class ServiceImplConfig {

    private ContextConfig contextConfig;

    private String pathTemplate;
    private String packageName; //service包名
    private List<String> imports;

    public void init() {
        ArrayList<String> imports = new ArrayList<>();
        imports.add("org.apache.dubbo.config.annotation.DubboService");
        imports.add("org.springframework.beans.BeanUtils");
        imports.add("javax.annotation.Resource");
        imports.add(contextConfig.getProjectBasePackageName() + ".common.config.YmlConfig");
        imports.add("com.baomidou.mybatisplus.service.impl.ServiceImpl");
        if (StringUtils.isNotEmpty(contextConfig.getModuleName())) {
            imports.add(contextConfig.getProjectBasePackageName() + ".mapper." + contextConfig.getModuleName() + "." + contextConfig.getEntityName() + "Mapper");
        } else {
            imports.add(contextConfig.getProjectBasePackageName() + ".mapper." + contextConfig.getEntityName() + "Mapper");
        }
        if (StringUtils.isNotEmpty(contextConfig.getModuleName())){
            imports.add(contextConfig.getProjectBasePackageName() + ".models.service." + contextConfig.getModuleName() + ".I" + contextConfig.getEntityName() + "Service");
            imports.add(contextConfig.getProjectBasePackageName() + ".models.bo." + contextConfig.getModuleName() + "." + contextConfig.getEntityName() + "SaveBO");
            imports.add(contextConfig.getProjectBasePackageName() + ".models.dto." + contextConfig.getModuleName() + "." + contextConfig.getEntityName() + "DTO");
        } else {
            imports.add(contextConfig.getProjectBasePackageName() + ".models.service.I" + contextConfig.getEntityName() + "Service");
            imports.add(contextConfig.getProjectBasePackageName() + ".models.bo." + contextConfig.getEntityName() + "SaveBO");
            imports.add(contextConfig.getProjectBasePackageName() + ".models.dto." + contextConfig.getEntityName() + "DTO");
        }
        this.imports = imports;


        //service模板生成路径

        if (StringUtils.isNotEmpty(contextConfig.getModuleName())) {
            this.packageName = contextConfig.getProjectBasePackageName() + ".service." + contextConfig.getModuleName();
        } else {
            this.packageName = contextConfig.getProjectBasePackageName() + ".service";
        }

        //serviceImpl模板生成路径
        if (StringUtils.isNotEmpty(contextConfig.getModuleName())) {
            this.pathTemplate = "\\" + contextConfig.getProjectName() + "-server\\src\\main\\java\\" + contextConfig.getProjectBasePackageName().replaceAll("\\.", "\\\\") + "\\service\\" + contextConfig.getModuleName() + "\\{}ServiceImpl.java";
        } else {
            this.pathTemplate = "\\" + contextConfig.getProjectName() + "-server\\src\\main\\java\\" + contextConfig.getProjectBasePackageName().replaceAll("\\.", "\\\\") + "\\service\\{}ServiceImpl.java";
        }
    }
}
